package com.vite.mybatisplus.util;


import com.vite.mybatisplus.context.page.PageContextHolder;
import com.vite.mybatisplus.context.page.model.ViteTableData;
import com.vite.mybatisplus.context.page.model.TableInfo;

import java.util.List;

public class PageUtils {
    public static <E> ViteTableData toTableInfo(List<E> list) {
        if (list instanceof TableInfo<E> tableInfo) {
            ViteTableData viteTableData = new ViteTableData();
            viteTableData.setData(list);
            viteTableData.setTotal(tableInfo.getTotal());
            return viteTableData;
        }
        ViteTableData viteTableData = new ViteTableData();
        viteTableData.setData(list);
        viteTableData.setTotal(-1L);
        return viteTableData;

    }

    public static void viteStartPage() {
        PageContextHolder.startPage();
        PageContextHolder.setPageInfo();
    }

}
